""" 欢迎关注 "码农架构" 微信公众号，热爱开源，拥抱开源。一个IT民工的技术之路经验分享。
    - 问题咨询 / 建议
    1.关注微信公众号 "码农架构" 后私信
    2.可发送邮件: li.shangzhi@aliyun.com
"""
# !/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2019/12/30 13:40
# @Author  : iByte

# Python3 批量修改JPG图片尺寸？
#
# 1.批量修改当前文件夹下的jpg图片到设置的尺寸
# 2.将修改后的图片移动到 new_img 这个文件夹下

from PIL import Image       # 处理图片模块
import os
import shutil               # 移动文件模块

x = input('请输入需要修改的尺寸，长：')
x = int(x)
y = input('请输入需要修改的尺寸，高：')
y = int(y)

now_path = os.getcwd()
new_path = os.mkdir(now_path + '\\' + 'new_img')
new_dir = now_path + '\\' + 'new_img'

# 修改图片大小
def smaller_img(x, y, path):
    path = str(path)
    old_img = Image.open(path)
    img_deal = old_img.resize((x, y), Image.ANTIALIAS)
    img_deal = img_deal.convert('RGB')
    img_deal.save('switch_' + file_name)
# 遍历文件夹下的文件，并判断是否是JPG文件
for file_name in os.listdir(now_path):
    files_path = now_path + '\\' + file_name
    if 'jpg' in files_path:
        smaller_img(x, y, files_path)
        # 遍历文件来判断是否是转换后的jpg文件
        for move_name in os.listdir(now_path):
            move_path = now_path + '\\' + move_name
            if 'switch' in move_path:
                shutil.move(move_path,new_dir)
            else:
                 print(move_path, '无须移动')
        print(file_name, 'switch success')
    else:
        print(file_name, 'is not img')
